package com.intellagric.service.traceability.Impl.trace_person;

import com.intellagric.common.pojo.LayuiDataGridResult;
import com.intellagric.common.pojo.ResponseMessage;
import com.intellagric.mapper.TracePersonTypeMapper;
import com.intellagric.pojo.TracePersonType;
import com.intellagric.pojo.TracePersonTypeExample;
import com.intellagric.service.traceability.trace_person.PersonTypeService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author: zhy
 * @email: 1269679946@qq.com
 * @Date: 2018/11/7
 * @DESC： 人员分类信息管理服务
 */
@Service
public class PersonTypeServiceImpl implements PersonTypeService {

    @Autowired
    private TracePersonTypeMapper personTypeMapper;

    /**
     * 添加人员分类信息
     * @param personType
     * @return ResponseMessage
     */
    @Override
    public ResponseMessage addPersonType(TracePersonType personType) {
        if (personTypeMapper.insertSelective(personType) == 1) {
            return ResponseMessage.success();
        } else {
            return ResponseMessage.fail();
        }
    }

    /**
     * 删除人员分类信息
     * @param id
     * @return ResponseMessage
     */
    @Override
    public ResponseMessage deletePersonType(String id) {
        if (personTypeMapper.deleteByPrimaryKey(id) == 1) {
            return ResponseMessage.success();
        } else {
            return ResponseMessage.fail();
        }
    }

    /**
     * 修改人员分类信息
     * @param personType
     * @return ResponseMessage
     */
    @Override
    public ResponseMessage editPersonType(TracePersonType personType) {
        if (personTypeMapper.updateByPrimaryKeySelective(personType)  == 1) {
            return ResponseMessage.success();
        } else {
            return ResponseMessage.fail();
        }
    }

    /**
     * 查询人员分类信息
     * @param id
     * @return TracePersonType
     */
    @Override
    public TracePersonType getPersonTypeById(String id) {
        return personTypeMapper.selectByPrimaryKey(id);
    }

    /**
     * 查询人员分类信息
     * @param parentId
     * @return TracePersonType
     */
    @Override
    public LayuiDataGridResult getPersonTypeByParentId(String parentId) {
        TracePersonTypeExample personTypeExample = new TracePersonTypeExample();
        personTypeExample.createCriteria().andParentIdEqualTo(parentId);
        List<TracePersonType> personTypeList = personTypeMapper.selectByExample(personTypeExample);
        return LayuiDataGridResult.success().add(personTypeList,personTypeList.size());
    }

    /**
     * 查询人员分类信息列表
     * @return TracePersonType
     */
    @Override
    public LayuiDataGridResult getPersonTypeList() {
        TracePersonTypeExample personTypeExample = new TracePersonTypeExample();
        personTypeExample.createCriteria().andIdIsNotNull();
        List<TracePersonType> personTypeList = personTypeMapper.selectByExample(personTypeExample);
        return LayuiDataGridResult.success().add(personTypeList,personTypeList.size());
    }

    /**
     * 根据创建者Id查询人员分类信息列表,并生成树形数据
     * @return TracePersonType
     */
    @Override
    public LayuiDataGridResult getPersonTypeTree(String createBy) {
        //查询所有的类别列表信息
        TracePersonTypeExample personTypeExample1 = new TracePersonTypeExample();
        personTypeExample1.createCriteria().andCreateByEqualTo(createBy);
        List<TracePersonType> personTypeList = personTypeMapper.selectByExample(personTypeExample1);

        //查询所有的父级类别列表
        TracePersonTypeExample personTypeExample2 = new TracePersonTypeExample();
        personTypeExample2.createCriteria().andParentIdEqualTo("0").andCreateByEqualTo(createBy);
        List<TracePersonType> rootpersonTypeList = personTypeMapper.selectByExample(personTypeExample2);

        for (TracePersonType personType : rootpersonTypeList) {
            personType.setChildren(getChild(personType.getId(), personTypeList));
        }
        return LayuiDataGridResult.success().add(rootpersonTypeList,rootpersonTypeList.size());
    }



    /**
     * 查询人员分类信息列表,并生成树形数据
     * @return TracePersonType
     */
    @Override
    public LayuiDataGridResult getPersonTypeTree() {
        List<TracePersonType> personTypeList = this.getPersonTypeList().getData();
        List<TracePersonType> rootpersonTypeList = this.getPersonTypeByParentId("0").getData();
        for (TracePersonType personType : rootpersonTypeList) {
            personType.setChildren(getChild(personType.getId(), personTypeList));
        }
        return LayuiDataGridResult.success().add(rootpersonTypeList,rootpersonTypeList.size());
    }

    private List<TracePersonType> getChild(String id, List<TracePersonType> personTypeList) {
        // 子菜单
        List<TracePersonType> childList = new ArrayList<>();

        // 遍历所有节点，将父菜单id与传过来的id比较
        for (TracePersonType personType : personTypeList) {
            if (personType.getParentId() != null) {   //所有非以及菜单的都要加入chidList中（递归）
                if (personType.getParentId().equals(id)) {
                    childList.add(personType);
                }
            }
        }

        // 把子菜单的子菜单再循环一遍
        for (TracePersonType personType : childList) {// 没有url子菜单还有子菜单（暂不考虑url的问题）
            // 递归
            personType.setChildren(getChild(personType.getId(), personTypeList));
        } // 递归退出条件
        if (childList.size() == 0) {
            return null;
        }
        return childList;
    }


}
